package com.demo.Main.FtpUtil;

import cn.hutool.core.io.FileUtil;
import cn.hutool.extra.ftp.Ftp;
import cn.hutool.extra.ftp.FtpMode;
import org.apache.commons.lang3.StringUtils;

import java.io.File;
import java.io.IOException;

/**
 * @author :zxc
 * @description :
 * @date :
 */
public class MyFtpUtil {

    /**
     *
     * @param remoteDir 上传的ftp目录
     * @param remoteFileName  保存到ftp服务器上的名称
     * @param localFile 本地文件全名称
     * @param ip
     * @param port
     * @param username
     * @param password
     * @return 成功则返回字符串:success
     */
    public static String upload(String remoteDir, String remoteFileName, String localFile, String ip, Integer port, String username, String password, FtpMode ftpMode) {
        if(StringUtils.isBlank(localFile)) {
            return "本地文件名称不能为空";
        }
        File lFile = FileUtil.file(localFile);
        if(!lFile.exists()) {
            return "本地文件不存在";
        }
        Ftp ftp = null;
        try {
            //匿名登录（无需帐号密码的FTP服务器）
            ftp = new Ftp(ip,port == null ? 21 : port,username,password);
            ftp.setMode(FtpMode.Passive);
            if(ftpMode != null) {
                ftp.setMode(ftpMode);
            }
            if(StringUtils.isBlank(remoteFileName)) {
                ftp.upload(remoteDir, lFile);
            } else {
                ftp.upload(remoteDir, remoteFileName, lFile);
            }
        } catch (Exception e) {
            return e.getMessage();
        } finally {
            //关闭连接
            try {
                if(ftp != null)  ftp.close();
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
        }
        return "success";
    }
}

